
#ifndef __EDNN_MAXPOOL_H__
#define __EDNN_MAXPOOL_H__



#include "ednn.h"



struct ednn_layer_maxpool 
{
    struct ednn_layer super;

    ednn_shape3d_t kernel;
    ednn_shape3d_t stride;
    ednn_shape3d_t padding;
    ednn_padding_type padding_type;
};
typedef struct ednn_layer_maxpool ednn_layer_maxpool_t;


struct ednn_layer *ednn_maxpool_layer(ednn_shape3d_t kernel, 
    ednn_shape3d_t stride, ednn_padding_type padding_type);

#define MaxPool(k, s, p) ednn_maxpool_layer(k, s, p)

#endif /* __EDNN_MAXPOOL_H__ */
